package 贪心;

/**
 * @description:
 * @author: 小白白
 * @create: 2021-08-26
 **/

public class No134加油站 {

    /**
     * 在一条环路上有 N 个加油站，其中第 i 个加油站有汽油 gas[i] 升。
     * 你有一辆油箱容量无限的的汽车，从第 i 个加油站开往第 i+1 个加油站需要消耗汽油 cost[i] 升。你从其中的一个加油站出发，开始时油箱为空。
     * 如果你可以绕环路行驶一周，则返回出发时加油站的编号，否则返回 -1。
     * <p>
     * 说明: 
     * 如果题目有解，该答案即为唯一答案。
     * 输入数组均为非空数组，且长度相同。
     * 输入数组中的元素均为非负数。
     * 示例 1:
     * 输入:
     * gas  = [1,2,3,4,5]
     * cost = [3,4,5,1,2]
     * 输出: 3
     * 解释:
     * 从 3 号加油站(索引为 3 处)出发，可获得 4 升汽油。此时油箱有 = 0 + 4 = 4 升汽油
     * 开往 4 号加油站，此时油箱有 4 - 1 + 5 = 8 升汽油
     * 开往 0 号加油站，此时油箱有 8 - 2 + 1 = 7 升汽油
     * 开往 1 号加油站，此时油箱有 7 - 3 + 2 = 6 升汽油
     * 开往 2 号加油站，此时油箱有 6 - 4 + 3 = 5 升汽油
     * 开往 3 号加油站，你需要消耗 5 升汽油，正好足够你返回到 3 号加油站。
     * 因此，3 可为起始索引。
     * 示例 2:
     * 输入:
     * gas  = [2,3,4]
     * cost = [3,4,3]
     * <p>
     * 输出: -1
     * 解释:
     * 你不能从 0 号或 1 号加油站出发，因为没有足够的汽油可以让你行驶到下一个加油站。
     * 我们从 2 号加油站出发，可以获得 4 升汽油。 此时油箱有 = 0 + 4 = 4 升汽油
     * 开往 0 号加油站，此时油箱有 4 - 3 + 2 = 3 升汽油
     * 开往 1 号加油站，此时油箱有 3 - 3 + 3 = 3 升汽油
     * 你无法返回 2 号加油站，因为返程需要消耗 4 升汽油，但是你的油箱只有 3 升汽油。
     * 因此，无论怎样，你都不可能绕环路行驶一周。
     */

    // normal写法
    public int canCompleteCircuit(int[] gas, int[] cost) {

        // 到当前地点会不会亏
        int[] arr = new int[gas.length];
        arr[0] = gas[0] - cost[0];

        int allCost = arr[0];

        for (int i = 1; i < arr.length; i++) {
            arr[i] = (gas[i] - cost[i]);
            allCost += arr[i];
        }

        if (allCost < 0) {
            return -1;
        }

        for (int i = 0; i < arr.length; i++) {
            int sum = 0;
            for (int j = 0; j < arr.length; j++) {
                sum += arr[((i + j) % arr.length)];
                if (sum < 0) {
                    break;
                }
            }
            if (sum >= 0) {
                return i;
            }
        }

        /**
         * 不好的写法,更好的是O(n)下方
         */
        return -1;
    }

    /**
     * 不管多少个站,都可以看成范围站,范围站一定不能小于0
     */
    public int canCompleteCircuitGood(int[] gas, int[] cost) {
        int currentGas = 0, allGas = 0;
        int result = 0;
        for (int i = 0; i < gas.length; i++) {
            currentGas += (gas[i] - cost[i]);
            allGas += (gas[i] - cost[i]);
            if (currentGas < 0) {
                currentGas = 0;
                result = i + 1;
            }
        }
        return allGas >= 0 ? result : -1;
    }

    public static void main(String[] args) {
        No134加油站 n = new No134加油站();
        int[] arr1 = {1, 1, 1, 10};
        int[] arr2 = {4, 4, 2, 1};
        int result = n.canCompleteCircuitGood(arr1, arr2);
        System.out.println(result);
    }

}
